from netfilterqueue import NetfilterQueue
from packet_correctness import *

ids = IDS('../NFs/ids_rules.csv')
packet_correctness_ids = PacketCorrectness(ids, 110, 'IDS')

_pkt = b'E\x00\x00X\x00\x01\x00\x00@\x06F\x13\xac\n\x01\x02$\x91b\xef\x1f\x90M\xf1\x00\x00\x00\x00\x00\x00\x00\x00P' \
       b'\x02 ' \
       b'\x00\xf6\xc6\x00\x00abcdefffffffffffffff\x00\x00\x00\x01\x00\x00\x00\x0b\x00\x00\x00n\\\xc3\xb6\x13\xfcDF\xe6' \
       b'\xad\xe6\xd826P\x83\xae '


def print_and_accept(pkt):
    # print(pkt)
    # out_pkt = packet_correctness_ids.process(_pkt)  # IDS
    # pkt.set_payload(out_pkt)
    pkt.accept()


queue = NetfilterQueue()
queue.bind(1, print_and_accept, max_len=8192)
try:
    queue.run()
except KeyboardInterrupt:
    queue.unbind()
